package com.nine.news.modular.filter;



import com.alibaba.fastjson.JSON;
import com.nine.news.constants.RedisKey;
import com.nine.news.constants.SystemConfig;
import com.nine.news.modular.annation.Token;
import com.nine.news.response.R;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.lang.reflect.Method;

/**
 * ━━━━━━Feri出没━━━━━━
 * 　　　┏┓　　　┏┓
 * 　　┏┛┻━━━┛┻┓
 * 　　┃　　　　　　 ┃
 * 　　┃　　　━　　　┃
 * 　　┃　┳┛　┗┳　  ┃
 * 　　┃　　　　　　 ┃
 * 　　┃　　　┻　　　┃
 * 　　┃　　　　　　　┃
 * 　　┗━┓　　　┏━┛
 * 　　　　┃　　　┃    邢哥的代码，怎么会，有bug呢，不可能！
 * 　　　　┃　　　┃
 * 　　　　┃　　　┗━━━┓
 * 　　　　┃　　　　　　　┣┓
 * 　　　　┃　　　　　　　┏┛
 * 　　　　┗┓┓┏━┳┓┏┛
 * 　　　　　┃┫┫　┃┫┫
 * 　　　　　┗┻┛　┗┻┛
 * ━━━━━━永无BUG!━━━━━━
 *
 * @Description:
 * @Author：邢朋辉
 * @Date: 2023/3/21 17:39
 */
@Component
public class TokenHandlerInterceptor implements HandlerInterceptor {
    @Resource
    private StringRedisTemplate template;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 如果不是映射到方法直接通过 带有@RequestMapping的方法
        if(!(handler instanceof HandlerMethod)){
            return true;
        }
        HandlerMethod handlerMethod=(HandlerMethod)handler;
        Method method=handlerMethod.getMethod();
        //检查是否有passtoken注释，有则跳过认证
        if (method.isAnnotationPresent(Token.class)) {
            //需要校验令牌 是否有效
            String token = request.getHeader(SystemConfig.HEADER_AUTH);// 从 http 请求头中取出 token
            if(StringUtils.hasLength(token)){
                if(template.hasKey(RedisKey.AUTH_USER_TOKEN+token)){
                    //解析令牌 uid 存储请求消息头
                    String uid=template.opsForValue().get(RedisKey.AUTH_USER_TOKEN+token);
                    request.setAttribute(SystemConfig.HEADER_UID,uid);
                    return true;
                }
            }
            //拦截 令牌无效
            response.setContentType("application/json;charset=UTF-8");
            response.getWriter().println(JSON.toJSONString(R.ERROR("令牌已失效，请重新登录！")));
            return false;
        }
        return true;
    }
}
